//
// Created by Administrator on 2021/4/13.
//

/*
给定一个二叉搜索树和一个目标结果，如果 BST 中存在两个元素且它们的和等于给定的目标结果，则返回 true。

案例 1:

输入:
5
/ \
  3   6
/ \   \
2   4   7

Target = 9

输出: True


案例 2:

输入:
5
/ \
  3   6
/ \   \
2   4   7

Target = 28

输出: False

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/two-sum-iv-input-is-a-bst
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。*/

#include <vector>
#include <queue>
#include <iostream>
#include <algorithm>
#include <numeric>

using namespace std;

struct TreeNode {
    int val; // 节点值
    TreeNode *left; // 左子节点
    TreeNode *right; // 右子节点

    TreeNode() : val(0), left(nullptr), right(nullptr) {}

    explicit TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}

    TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
};

class Solution {
private:
    vector<int> v;

public:
    bool findTarget(TreeNode *root, int k) {
        // 给了一个二叉搜索树，用中序遍历变成一个递增数列
        // 再对数列从头尾闭合搜索
        // dfs 完成中序遍历
        dfs(root);
        // 得到递增数列
//        for (auto &x:v)cout << x << endl;
        // 从头尾搜索
        auto left = v.begin();
        auto right = v.end() - 1;
        while (left < right) {
            int sumVal = *left + *right;
            if (sumVal == k) return true;
            else if (sumVal > k) right--;
            else left++;
        }
        return false;
    }

    void dfs(TreeNode *root) {
        if (root == nullptr) return;
        dfs(root->left);
        v.push_back(root->val);
        dfs(root->right);
    }
};

class Solution2  // 优秀解答
{
public:
    bool findTarget(TreeNode *root, int k) {
        vector<int> nums;
        dfs(root, nums);
        int i = 0, j = (int) nums.size() - 1;
        while (i < j) {
            int sum = nums[i] + nums[j];
            if (sum == k)
                return true;
            else if (sum > k)
                j--;
            else
                i++;
        }
        return false;
    }

    void dfs(TreeNode *root, vector<int> &nums) {
        if (!root)
            return;
        dfs(root->left, nums);
        nums.push_back(root->val);
        dfs(root->right, nums);
    }
};

int main() {
    auto t6 = TreeNode(7);
    auto t5 = TreeNode(4);
    auto t4 = TreeNode(2);
    auto t3 = TreeNode(6, nullptr, &t6);
    auto t2 = TreeNode(3, &t4, &t5);
    auto t1 = TreeNode(5, &t2, &t3);
    Solution sol;
    cout << (sol.findTarget(&t1, 9) ? "true" : "false");
    return 0;
}